package countKDifference;

import java.util.HashMap;
import java.util.Map;

public class Solution {
    public int countKDifference(int[] nums, int k) {
//        int count = 0;
//        Map<Integer,Integer> map = new HashMap<>();
//        for (int i = 0; i < nums.length; i++) {
//            count += map.getOrDefault(nums[i] - k,0);
//            count += map.getOrDefault(nums[i] + k ,0);
//            map.put(nums[i],map.getOrDefault(nums[i],0)+1);
//        }
//        return count;

        int ans = 0;
        int[] marks = new int[101+k];
        for (int i = 0; i < nums.length; i++) {
            if( nums[i] < k){
                continue;
            }
            ans += marks[nums[i] - k];
            ans += marks[nums[i] + k];
            marks[nums[i]]++;
        }
        return ans;
    }
}
